from baidu.base import Base
from selenium.webdriver.common.by import By

class Page(Base):
    # 定位器
    # ‘新闻’
    xinwen_loc = (By.XPATH,'//*[@id="s-top-left"]/a[1]')
    # 搜索框定位元素
    sousuo_loc = (By.ID,'chat-textarea')
    # 搜索按钮
    sousuo_btn_loc = (By.ID,'chat-submit-button')
    # 相关搜索
    xiangguan_loc = (By.XPATH,'//*[@id="rs_new"]/div/div')
    # 第二页
    two_loc = (By.XPATH,'//*[@id="page"]/div/a[1]/span')
    # 网站的第一个标题 定位元素
    title_loc = (By.XPATH,'//*[@id="3001"]/div/div/div/div[1]/h3/div/a')


    # 方法封装

    # ‘新闻’判断进入百度页面
    def dy_xinwen(self):
        """‘新闻’判断进入百度页面"""
        return self.get_text(self.xinwen_loc)

    def sousuo(self):
        """搜索框"""
        self.send_keys(self.sousuo_loc,'我喜欢软件测试')

    def sousuo_btn(self):
        """搜索按钮"""
        self.click(self.sousuo_btn_loc)

    def dy_xiangguan(self):
        return self.get_text(self.xiangguan_loc)

    def two(self):
        self.click(self.two_loc)

    def title(self):
        return self.get_text(self.title_loc)